from app.schema.base import UserBaseSchema,InfoSchema,RespDataSchema
from pydantic import BaseModel


class Token(BaseModel):
    access_token: str
    token_type: str='Bearer'
    refresh_token:str

class AuthDataSchema(RespDataSchema):
    code: int = 200
    message: str = 'SUCCESS'
    data:Token

class LoginInputSchema(BaseModel):
    code: str
    
class UserInfoSchema(InfoSchema):
    id:int=0
    avatar:str=None
    
class RespUserSchema(RespDataSchema):
    data:dict={}
    
class TokenData(BaseModel):
    id:int
    name: str
    avatar: str

class AiInfoSchema(BaseModel):
    ai_id:int